home *** CD-ROM | disk | FTP | other *** search
/ Total Network Tools 2002 / NextStepPublishing-TotalNetworkTools2002-Win95.iso / Archive / Misc Servers / Zope.exe / DBI_DB.PY < prev    next >
Encoding:
Python Source  |  2000-05-30  |  8.1 KB  |  223 lines
  1. ##############################################################################
  2. # 
  3. # Zope Public License (ZPL) Version 1.0
  4. # -------------------------------------
  5. # 
  6. # Copyright (c) Digital Creations.  All rights reserved.
  7. # 
  8. # This license has been certified as Open Source(tm).
  9. # 
  10. # Redistribution and use in source and binary forms, with or without
  11. # modification, are permitted provided that the following conditions are
  12. # met:
  13. # 
  14. # 1. Redistributions in source code must retain the above copyright
  15. #    notice, this list of conditions, and the following disclaimer.
  16. # 
  17. # 2. Redistributions in binary form must reproduce the above copyright
  18. #    notice, this list of conditions, and the following disclaimer in
  19. #    the documentation and/or other materials provided with the
  20. #    distribution.
  21. # 
  22. # 3. Digital Creations requests that attribution be given to Zope
  23. #    in any manner possible. Zope includes a "Powered by Zope"
  24. #    button that is installed by default. While it is not a license
  25. #    violation to remove this button, it is requested that the
  26. #    attribution remain. A significant investment has been put
  27. #    into Zope, and this effort will continue if the Zope community
  28. #    continues to grow. This is one way to assure that growth.
  29. # 
  30. # 4. All advertising materials and documentation mentioning
  31. #    features derived from or use of this software must display
  32. #    the following acknowledgement:
  33. # 
  34. #      "This product includes software developed by Digital Creations
  35. #      for use in the Z Object Publishing Environment
  36. #      (http://www.zope.org/)."
  37. # 
  38. #    In the event that the product being advertised includes an
  39. #    intact Zope distribution (with copyright and license included)
  40. #    then this clause is waived.
  41. # 
  42. # 5. Names associated with Zope or Digital Creations must not be used to
  43. #    endorse or promote products derived from this software without
  44. #    prior written permission from Digital Creations.
  45. # 
  46. # 6. Modified redistributions of any form whatsoever must retain
  47. #    the following acknowledgment:
  48. # 
  49. #      "This product includes software developed by Digital Creations
  50. #      for use in the Z Object Publishing Environment
  51. #      (http://www.zope.org/)."
  52. # 
  53. #    Intact (re-)distributions of any official Zope release do not
  54. #    require an external acknowledgement.
  55. # 
  56. # 7. Modifications are encouraged but must be packaged separately as
  57. #    patches to official Zope releases.  Distributions that do not
  58. #    clearly separate the patches from the original work must be clearly
  59. #    labeled as unofficial distributions.  Modifications which do not
  60. #    carry the name Zope may be packaged in any form, as long as they
  61. #    conform to all of the clauses above.
  62. # 
  63. # 
  64. # Disclaimer
  65. # 
  66. #   THIS SOFTWARE IS PROVIDED BY DIGITAL CREATIONS ``AS IS'' AND ANY
  67. #   EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  68. #   IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  69. #   PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL DIGITAL CREATIONS OR ITS
  70. #   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  71. #   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  72. #   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  73. #   USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  74. #   ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  75. #   OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  76. #   OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  77. #   SUCH DAMAGE.
  78. # 
  79. # 
  80. # This software consists of contributions made by Digital Creations and
  81. # many individuals on behalf of Digital Creations.  Specific
  82. # attributions are listed in the accompanying credits file.
  83. # 
  84. ##############################################################################
  85. __doc__='''short description
  86.  
  87.  
  88. $Id: dbi_db.py,v 1.8 2000/05/30 15:46:55 brian Exp $'''
  89. #     Copyright 
  90. #
  91. #       Copyright 1997 Digital Creations, Inc, 910 Princess Anne
  92. #       Street, Suite 300, Fredericksburg, Virginia 22401 U.S.A. All
  93. #       rights reserved. 
  94. # 
  95. __version__='$Revision: 1.8 $'[11:-2]
  96.  
  97. import string, sys
  98. from string import strip, split, find, join
  99.  
  100. failures=0
  101. calls=0
  102.  
  103. nonselect_desc=[
  104.     ('Query',  'STRING', 62, 62, 0, 0, 1),
  105.     ('Status', 'STRING', 12, 12, 0, 0, 1),
  106.     ('Calls',  'STRING', 12, 12, 0, 0, 1),
  107.     ]
  108.  
  109. class DB:
  110.  
  111.     _p_oid=_p_changed=_registered=None
  112.  
  113.     defs={'STRING':'s', 'NUMBER':'n', 'DATE':'d'}
  114.  
  115.     def Database_Connection(self, string):
  116.         # Create a dbi-compatible database connection
  117.         raise 'ImplementedBySubclass', (
  118.             'attempt to create a database connection for an abstract dbi')
  119.  
  120.     Database_Error='Should be overriden by subclass'
  121.  
  122.     def __init__(self,connection):
  123.         self.connection=connection
  124.         db=self.db=self.Database_Connection(connection)
  125.         self.cursor=db.cursor()
  126.         
  127.     def str(self,v, StringType=type('')):
  128.         if v is None: return ''
  129.         r=str(v)
  130.         if r[-1:]=='L' and type(v) is not StringType: r=r[:-1]
  131.         return r
  132.  
  133.     def __inform_commit__(self, *ignored):
  134.         self._registered=None
  135.         self.db.commit()
  136.  
  137.     def __inform_abort__(self, *ignored):
  138.         self._registered=None
  139.         self.db.rollback()
  140.  
  141.     def register(self):
  142.         if self._registered: return
  143.         get_transaction().register(self)
  144.         self._registered=1
  145.         
  146.  
  147.     def query(self,query_string, max_rows=9999999):
  148.         global failures, calls
  149.         calls=calls+1
  150.         try:
  151.             c=self.cursor
  152.             self.register()
  153.             queries=filter(None, map(strip,split(query_string, '\0')))
  154.             if not queries: raise 'Query Error', 'empty query'
  155.             if len(queries) > 1:
  156.                 result=[]
  157.                 for qs in queries:
  158.                     r=c.execute(qs)
  159.                     if r is None: raise 'Query Error', (
  160.                         'select in multiple sql-statement query'
  161.                         )
  162.                     result.append((qs, str(`r`), calls))
  163.                 desc=nonselect_desc
  164.             else:
  165.                 query_string=queries[0]
  166.                 r=c.execute(query_string)
  167.                 if r is None:
  168.                     result=c.fetchmany(max_rows)
  169.                     desc=c.description
  170.                 else:
  171.                     result=((query_string, str(`r`), calls),)
  172.                     desc=nonselect_desc
  173.             failures=0
  174.             c.close()
  175.         except self.Database_Error, mess:
  176.             c.close()
  177.             self.db.rollback()
  178.             failures=failures+1
  179.             if ((find(mess,": invalid") < 0 and
  180.                  find(mess,"PARSE") < 0) or
  181.                 # DBI IS stupid
  182.                 find(mess,
  183.                      "Error while trying to retrieve text for error") > 0
  184.                 or
  185.                 # If we have a large number of consecutive failures,
  186.                 # our connection is probably dead.
  187.                 failures > 100
  188.                 ):
  189.                 # Hm. maybe the db is hosed.  Let's try once to restart it.
  190.                 failures=0
  191.                 c.close()
  192.                 self.db.close()
  193.                 db=self.db=self.Database_Connection(self.connection)
  194.                 self.cursor=db.cursor()
  195.                 c=self.cursor
  196.                 c.execute(query_string)
  197.                 result=c.fetchall()
  198.                 desc=c.description
  199.             else:
  200.                 raise sys.exc_info()
  201.         
  202.         if result:
  203.             result=join(
  204.                     map(
  205.                         lambda row, self=self:
  206.                         join(map(self.str,row),'\t'),
  207.                         result),
  208.                     '\n')+'\n'
  209.         else:
  210.             result=''
  211.  
  212.         return (
  213.             "%s\n%s\n%s" % (
  214.                 join(map(lambda d: d[0],desc), '\t'),
  215.                 join(
  216.                     map(
  217.                         lambda d, defs=self.defs: "%d%s" % (d[2],defs[d[1]]),
  218.                         desc),
  219.                     '\t'),
  220.                 result,
  221.                 )
  222.             )
  223.